package com.yunxi.module.system.service.permission.impl;

import cn.hutool.core.collection.CollUtil;
import com.google.common.annotations.VisibleForTesting;
import com.google.common.collect.Lists;
import com.yunxi.framework.common.exception.util.ServiceExceptionUtil;
import com.yunxi.framework.common.util.object.BeanUtils;
import com.yunxi.module.system.enums.constant.ErrorCodeConstants;
import com.yunxi.module.system.enums.permission.MenuTypeEnum;
import com.yunxi.module.system.mapper.permission.MenuMapper;
import com.yunxi.module.system.model.permission.MenuDO;
import com.yunxi.module.system.redis.RedisKeyConstants;
import com.yunxi.module.system.service.permission.MenuService;
import com.yunxi.module.system.service.permission.PermissionService;
import com.yunxi.module.system.service.tenant.TenantService;
import com.yunxi.module.system.vo.admin.permission.menu.MenuListReqVO;
import com.yunxi.module.system.vo.admin.permission.menu.MenuSaveReqVO;
import lombok.extern.slf4j.Slf4j;
import org.springframework.cache.annotation.CacheEvict;
import org.springframework.cache.annotation.Cacheable;
import org.springframework.context.annotation.Lazy;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import javax.annotation.Resource;
import java.util.Collection;
import java.util.List;

import static com.yunxi.framework.common.util.collection.CollectionUtils.convertList;

/**
 * <p>
 * 菜单 Service 接口实现类
 * </p>
 *
 * @author lidy
 * @since 2024-05-07
 */
@Service
@Slf4j
public class MenuServiceImpl implements MenuService {

    @Resource
    private MenuMapper menuMapper;

    @Resource
    @Lazy // 延迟，避免循环依赖报错
    private TenantService tenantService;

    @Resource
    private PermissionService permissionService;

    /**
     * 获得权限对应的菜单编号数组
     *
     * @param permission 权限标识
     * @return 数组
     */
    @Override
    @Cacheable(value = RedisKeyConstants.PERMISSION_MENU_ID_LIST, key = "#permission")
    public List<Long> getMenuIdListByPermissionFromCache(String permission) {
        List<MenuDO> menus = menuMapper.selectListByPermission(permission);
        return convertList(menus, MenuDO::getId);
    }

    /**
     * 获得全部菜单列表
     *
     * @return 全部菜单列表
     */
    @Override
    public List<MenuDO> getMenuList() {
        return menuMapper.selectList();
    }

    /**
     * 获得全部菜单列表
     *
     * @param menuIds 菜单编号集合
     * @return 全部菜单列表
     */
    @Override
    public List<MenuDO> getMenuList(Collection<Long> menuIds) {
        // 当菜单编号集合为空时，返回一个空集合
        if (CollUtil.isEmpty(menuIds)) {
            return Lists.newArrayList();
        }
        return menuMapper.selectBatchIds(menuIds);
    }

    /**
     * 获得菜单列表
     *
     * @param reqVO 查询条件
     * @return 菜单列表
     */
    @Override
    public List<MenuDO> getMenuList(MenuListReqVO reqVO) {
        return menuMapper.selectList(reqVO);
    }

    /**
     * 获得菜单信息
     *
     * @param menuId 菜单编号
     * @return 菜单信息
     */
    @Override
    public MenuDO getMenu(Long menuId) {
        return menuMapper.selectById(menuId);
    }

    /**
     * 基于租户，筛选菜单列表
     * 注意：如果是系统租户，返回的也是全菜单
     *
     * @param reqVO 筛选条件
     * @return 菜单列表
     */
    @Override
    public List<MenuDO> getMenuListByTenant(MenuListReqVO reqVO) {
        // 获取全菜单
        List<MenuDO> menuList = getMenuList(reqVO);
        // 如果开启了多租户，过滤掉未开通的菜单
        tenantService.handleTenantMenu(menuIds -> menuList.removeIf(menu -> !CollUtil.contains(menuIds, menu.getId())));
        return menuList;
    }

    /**
     * 创建菜单
     *
     * @param saveReqVO 菜单信息
     * @return 菜单编号
     */
    @Override
    @CacheEvict(value = RedisKeyConstants.PERMISSION_MENU_ID_LIST, key = "#saveReqVO.permission",
            condition = "#saveReqVO.permission != null")
    public Long createMenu(MenuSaveReqVO saveReqVO) {
        // 校验父菜单是否存在
        validateParentMenu(saveReqVO.getParentId(), null);

        // 校验菜单信息
        validateMenu(saveReqVO.getParentId(), saveReqVO.getName(), null);

        // 保存到数据库
        MenuDO menu = BeanUtils.toBean(saveReqVO, MenuDO.class);
        initMenuProperty(menu);
        menuMapper.insert(menu);

        // 返回
        return menu.getId();
    }

    /**
     * 更新菜单信息
     * allEntries 清空所有缓存，因为 permission 如果变更，涉及到新老两个 permission。直接清理，简单有效
     *
     * @param updateReqVO 菜单信息
     */
    @Override
    @CacheEvict(value = RedisKeyConstants.PERMISSION_MENU_ID_LIST, allEntries = true)
    public void updateMenu(MenuSaveReqVO updateReqVO) {
        // 校验菜单是否存在
        if (menuMapper.selectById(updateReqVO.getId()) == null) {
            throw ServiceExceptionUtil.exception(ErrorCodeConstants.MENU_NOT_EXISTS);
        }

        // 校验父菜单
        validateParentMenu(updateReqVO.getParentId(), updateReqVO.getId());

        // 校验菜单信息
        validateMenu(updateReqVO.getParentId(), updateReqVO.getName(), updateReqVO.getId());

        // 更新数据
        MenuDO menuDO = BeanUtils.toBean(updateReqVO, MenuDO.class);
        initMenuProperty(menuDO);
        menuMapper.updateById(menuDO);
    }

    /**
     * 删除菜单
     * allEntries 清空所有缓存，因为 permission 如果变更，涉及到新老两个 permission。直接清理，简单有效
     *
     * @param menuId 菜单编号
     */
    @Override
    @Transactional
    @CacheEvict(value = RedisKeyConstants.PERMISSION_MENU_ID_LIST, allEntries = true)
    public void deleteMenu(Long menuId) {
        // 校验是否还有子菜单
        if (menuMapper.selectCountByParentId(menuId) > 0) {
            throw ServiceExceptionUtil.exception(ErrorCodeConstants.MENU_EXISTS_CHILDREN);
        }

        // 校验要删除的菜单是否存在
        if (menuMapper.selectById(menuId) == null) {
            throw ServiceExceptionUtil.exception(ErrorCodeConstants.MENU_NOT_EXISTS);
        }

        // 标记删除
        menuMapper.deleteById(menuId);

        // 删除授予给角色的权限
        permissionService.processMenuDeleted(menuId);
    }

    /**
     * 初始化菜单的通用属性
     * 例如：只有目录或者菜单类型的菜单，才设置 icon
     *
     * @param menu 菜单信息
     */
    private void initMenuProperty(MenuDO menu) {
        // 菜单为按钮类型时，无需 component、icon、path 属性，进行置空
        if (MenuTypeEnum.BUTTON.getType().equals(menu.getType())) {
            menu.setComponent("");
            menu.setComponentName("");
            menu.setIcon("");
            menu.setPath("");
        }
    }

    /**
     * 校验菜单是否合法
     * 相同父菜单下，不能有相同的菜单
     *
     * @param parentId 父菜单编号
     * @param name     菜单名称
     * @param menuId   菜单编号
     */
    private void validateMenu(Long parentId, String name, Long menuId) {
        MenuDO menuDO = menuMapper.selectByParentIdAndName(parentId, name);
        if (menuDO == null) {
            return;
        }

        // 如果 menuId 为空，说明不用比较是否为同一个菜单
        if (menuId == null) {
            throw ServiceExceptionUtil.exception(ErrorCodeConstants.MENU_NAME_DUPLICATE);
        }
        if (!menuId.equals(menuDO.getId())) {
            throw ServiceExceptionUtil.exception(ErrorCodeConstants.MENU_NAME_DUPLICATE);
        }
    }

    /**
     * 校验父菜单是否合法
     * 1.不能设置自己为父菜单
     * 2.父菜单不存在
     * 3.父菜单必须是 {@link MenuTypeEnum#MENU} 或 {@link MenuTypeEnum#DIR} 类型
     *
     * @param parentId 父菜单编号
     * @param childId  当前菜单编号
     */
    @VisibleForTesting
    private void validateParentMenu(Long parentId, Long childId) {
        if (parentId == null || MenuDO.ID_ROOT.equals(parentId)) {
            return;
        }

        // 不能设置自己为父菜单
        if (parentId.equals(childId)) {
            throw ServiceExceptionUtil.exception(ErrorCodeConstants.MENU_PARENT_ERROR);
        }

        // 父菜单不存在
        MenuDO menu = menuMapper.selectById(parentId);
        if (menu == null) {
            throw ServiceExceptionUtil.exception(ErrorCodeConstants.MENU_PARENT_NOT_EXISTS);
        }

        // 父菜单必须是目录或者菜单类型
        if (!MenuTypeEnum.DIR.getType().equals(menu.getType())
                && !MenuTypeEnum.MENU.getType().equals(menu.getType())) {
            throw ServiceExceptionUtil.exception(ErrorCodeConstants.MENU_PARENT_NOT_DIR_OR_MENU);
        }
    }

}
